How to add type hints for an inner or nested function inside the stub file